@echo off
echo 正在清理Docker环境并构建Ubuntu桌面环境...
echo.

REM 检查Docker是否运行
docker info >nul 2>&1
if %errorlevel% neq 0 (
    echo 错误: Docker未运行，请先启动Docker Desktop
    pause
    exit /b 1
)

echo 步骤1: 清理Docker系统...
echo 清理未使用的镜像...
docker image prune -f

echo 清理未使用的容器...
docker container prune -f

echo 清理未使用的卷...
docker volume prune -f

echo 清理未使用的网络...
docker network prune -f

echo 清理构建缓存...
docker builder prune -f

echo.
echo 步骤2: 检查可用磁盘空间...
docker system df

echo.
echo 步骤3: 构建Docker镜像...
echo 选择构建版本:
echo 1. 标准版本 (包含更多应用程序)
echo 2. 轻量级版本 (推荐，占用空间更少)
echo.
set /p choice="请输入选择 (1 或 2): "

if "%choice%"=="1" (
    echo 构建标准版本...
    docker-compose build --no-cache
    if %errorlevel% neq 0 (
        echo 构建失败，建议尝试轻量级版本
        pause
        exit /b 1
    )
    echo 启动标准版本...
    docker-compose up -d
) else if "%choice%"=="2" (
    echo 构建轻量级版本...
    docker-compose -f docker-compose.lightweight.yml build --no-cache
    if %errorlevel% neq 0 (
        echo 构建失败，请检查错误信息
        pause
        exit /b 1
    )
    echo 启动轻量级版本...
    docker-compose -f docker-compose.lightweight.yml up -d
) else (
    echo 无效选择，默认使用轻量级版本...
    docker-compose -f docker-compose.lightweight.yml build --no-cache
    docker-compose -f docker-compose.lightweight.yml up -d
)

if %errorlevel% neq 0 (
    echo 启动失败，请检查错误信息
    pause
    exit /b 1
)

echo.
echo Ubuntu桌面环境已启动！
echo.
echo 查看容器状态: docker-compose ps
echo 查看日志: docker-compose logs -f
echo 停止服务: docker-compose down
echo.
pause 